# !/usr/bin/env python
# -*-coding:utf-8 -*-

"""
# File       : BiLSTM.py
# Time       ：2021/3/8 17:40
# Author     ：author name
# version    ：python 3.6
# Description：
"""
import tensorflow as tf

batch_size=128
num_hiddens=0
units=64

input_dim = 28  # 必须和特征的列数目相等

class BiLSTM:
    def __init__(self):
        self.batch_size=batch_size
        self.num_hiddens=num_hiddens
        self.units=units
        pass

    def bilstm(self):

        lstm_cell_fw=tf.keras.layers.LSTMCell(units=units,dropout=0.5)
        lstm_cell_bw=tf.keras.layers.LSTMCell(units=units,dropout=0.5)

        tf.keras.layers.Dense(1)
